package cn.itsource.config;


import cn.dev33.satoken.reactor.filter.SaReactorFilter;
import cn.dev33.satoken.router.SaRouter;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.util.SaResult;
import cn.itsource.pojo.PermissionMsg;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.data.redis.core.RedisTemplate;

import java.util.Collections;
import java.util.List;

/**
 * [Sa-Token 权限认证] 配置类 
 * @author click33
 */
@Configuration
public class GatewaySaTokenConfigure {
    @Autowired
    private RedisTemplate<String,Object> redisTemplate;

    // 注册 Sa-Token全局过滤器 
    @Bean
    public SaReactorFilter getSaReactorFilter() {
        return new SaReactorFilter()
            // 拦截地址 
            .addInclude("/**")    /* 拦截全部path */
            // 开放地址 
            .addExclude("/app/driver/register")
            .addExclude("/romate/api/login")
            // 鉴权方法：每次访问进入 
            .setAuth(obj -> {
                // 登录校验 -- 拦截所有路由，并排除/user/doLogin 用于开放登录 
                SaRouter.match("/**", "/app/login/*", r -> StpUtil.checkLogin());
                
                // 权限认证 -- 不同模块, 校验不同权限

                PermissionMsg permissionMsg = (PermissionMsg) redisTemplate.opsForValue().get("permissionMsg");
                List<PermissionMsg> list = Collections.singletonList(permissionMsg);
                for (PermissionMsg msg : list) {
                    SaRouter.match( msg.getResource(), r -> StpUtil.checkPermission(msg.getSn()));
                }
                
                // 更多匹配 ...  */
            })
            // 异常处理方法：每次setAuth函数出现异常时进入 
            .setError(e -> {
                return SaResult.error(e.getMessage());
            })
            ;
    }
}
